More pythonic if statements