Create BusScheduleViewModel.kt

app/src/main/java/com/example/busschedule/viewmodels/BusScheduleViewModel.kt

@@ -0,0 +1,15 @@
+class BusScheduleViewModel(private val scheduleDao: ScheduleDao): ViewModel() {
+fun fullSchedule(): List<Schedule> = scheduleDao.getAll()
+
+fun scheduleForStopName(name: String): List<Schedule> = scheduleDao.getByStopName(name)
+class BusScheduleViewModelFactory(
+   private val scheduleDao: ScheduleDao
+) : ViewModelProvider.Factory {
+}
+override fun <T : ViewModel> create(modelClass: Class<T>): T {
+       if (modelClass.isAssignableFrom(BusScheduleViewModel::class.java)) {
+           @Suppress("UNCHECKED_CAST")
+           return BusScheduleViewModel(scheduleDao) as T
+       }
+       throw IllegalArgumentException("Unknown ViewModel class")
+   }