Update python script per review comments

Wrap to 80 cols, introduce main() function and use argparse instead of
optparse.