Fixed a minor bug with a print statement and an extraneous return