現在のステートメントを
if __name__ == '__main__:'
式で囲みます。