Make the code-statement polymorphic

I figured out how to make the code-statement polymorphic and changed the return-statement example to use this capability.