Python Logical Statements Math