Updated program to use proper Python style and accept arguments and lots of other nicel ittle changes
More tidying