python if not in one line