Added a model to be more realistic