Working on simple code generation