I forked their repo and make some tweaks.