minor assignment2 updates