Part of Data Science Course Assignment