For Data Science Toolkit assignment