removed the print statements and cleaned up unnecessary code.