Fix bugs. update readme