Changed the model to a simpler one