Implemented ViewModelTests for UI (3 of 3 tests are green)