Minnor changes for assignment #2