Update readme and minor details of the script