Problem 2

Still trying to make the function works.