Fix code, update readme