Fixed missing model code