CSE22125's submissions for Write a function