update readme
cleanup code