Minor changes for Assignment 1