Fixing up script to adhere to pep8 style guidelines.

Signed-off-by: Rob Dobson <rob@rdobson.co.uk>