CB_EN_U4CSE20433's submissions for Write a function