Added whole solution

After class review