Add some model code