small edits to code and README