Removed print statements and other unnecessary stuff.