Remove useless math import.
Add a solution to problem 2.