Use __init__ == '__main__' pattern in script

Use the typical __init__ == '__main__' pattern to determine if the code
is being run as a script, and only execute the main code if true. This
makes the code more portable to other projects.

Signed-off-by: Grant Likely <grant.likely@arm.com>