updating README, minor code changes