Used inheritance to make the creation of different models ...better?