Fixing up script to adhere to pep8 style guidelines.

Fixing up various violations of the pep8 guidelines.

Signed-off-by: Rob Dobson <rob@rdobson.co.uk>