Remade viewmodel and stuff