Add model code