minor changes to the code and readme