Removed print statement and added a few comments