Add comments, minor tidying, README.