Assignment 2: added a few extra comments