updating script to conform with python style