As you can see the type of the class variables is generic, but you can improve the generator to understand if a variable should contain a number or a function.