Added if __name__ == "__main__" to the script.