Attempted solution for Problem 2