Added some comments

Just to neaten up the script and make it easier to read etc.