edit to readme and a little code patch