Add "return", "yield" and "raise" expressions handling