Fix model code to be in code