Assignment 2 solution

Updated stub file to meet requirements for Assignment 2