Cleaned code and deleted print statements