Initial attempt: Programming Assignment 2