CB_EN_U4CSE20301's submissions for Write a function