Completing assignment for data science toolkit