Removed prints, pep8 some shit and change `is` to ==