fix bug and rewrite readme