improve assignment type checking, fix MultiLine and Return