CB_EN_U4_CSE20322's submissions for Write a function