CSE22117's submissions for Write a function