minor cleanup code and update readme