updates for Prog Assignment 2