Added starter code for code gen