WIP: Programming Assignment 2: first attempt