still working on the model