Updated with solutions to Session 1 assignment

Hopefully I've done this right... the tests.py script seemed to pass.