Use `if __name__` and fix whitespace.

You can use `if __name__ == '__main__'` to check if your module is being
imported or run as a script. This allows your module to be used externally,
tested, etc.