Improve model remove info