update: readme and small improvement