Thank you very much for posting this very practical implementation. I am very glad to inform you that I used it on my own medium post here :
Written by Emad Ezzeldin ,Sr. DataScientist@UnitedHealthGroup
5 years Data Scientist and a MSc from George Mason University in Data Analytics. I enjoy experimenting with Data Science tools. emad.ezzeldin4@gmail.com