Adding code generator code (In Progress)