Added tests for ViewModel