Programming assignment 1 code changes