add return exprssion directly
refactor
update readme