- support return statement.
- fix some bugs.