Final challenge of the Supervised ML models module