Adding my program assignment 2