CSE_20311's submissions for Write a function