Completed Problem 2

- Probably completed it. Should work...