Initial crack at Assignment 2