Tidy script and comments
Improve readme