Created as part of Data Scientist course project