update readme
some small code changes