Added model code