Remove 'if __name__...'