Edits to Assignment 2 stub

I edited this stub forked from the course github account to include the
code required for the assignment