My first attempt at the Assignment 2