Problem #2

Solution update