Modified to declare model explicitly