[python] If statement for strings in python?