Basic exercise for the data science course