Assignment 3 changes. Added comments too.