Support for Return Statements in Custom Code Action!