CB_EN_U4CSE20366's submissions for Write a function