Added my solution to the assignment