Awesome medium AI

Hieu did an awesome edit to the medium AI, now it works great