CBENU4CSE20441_'s submissions for Write a function