Migrate SetupController to ViewModel