a smidge more updates to the code generation