[CODEGEN] Correct bug with case generation
[CODE] Add RETURN instruction code generation