refactoring
add VisitReturnNode