Fixed problem 2 -- helps to actually follow through and start learning Python