CB_EN_U4CSE21357's submissions for Write a function