Removed some print statements and added a few comments