Add README and tidy up comments