Added custom model for testing