Improved feedback in automated grading of data science assignments