CSE22141's submissions for Write a function