tided up comments in the README and code file