Python - Same if statement multiple times in one code block