Code generation working great so far