Data Science diploma project