Created as part of the course project for "Data scientist's Toolbox"