minor changes in code and readme