Solved problem 3

It might be gross