Yizhou Zha Assignment 2 Version 1.1

Added some comments