fix code && update readme