Update data science solution

Add a newline here. This code is crowded. Try to separate different "paragraphs" of code with a newline.