Refactored for Python 2.7

Changed class names to better illustrate ideas covered in training,
changed imports for Python 2.7