My Gorn/Frostford mod has code for that.
It was unfortunately one of those code changes that got bigger and bigger in order to fix stuff, and the original code needed changing in a few different files, so it might not be easy to pick out just the changes you want.