Add simple class code generation