update readme and more comments