Changed Assignment code shell to complete Assignment 2