For data scientist assignment