Adding problem 3. It's really painful writing math in docstring format