Changed some logic, model doesnt work