Made if statements more python-y