Removed print statements and added some comments