Prog assignment 2