Clean up viewmodel