Assignment 2 updates

updated comments