Solution

cleaned up solution