Adding custom model code