Removed print statements and added comments