CSE22158's submissions for Write a function