Change View Model to work