CSE22121's submissions for Write a function