tidy code, update readme