Updates to programming assignmnet 2