Added viewmodel and got it to load