Added a better example to the if name = __main__ section.