Added final model test