CSE22124's submissions for Write a function