created as part of Data Scientist hw