Improved the Model