Refactoring the data science solution

Line is too long. [84/80]