Created as part of course project for Data Science