Didn't see this until just now, but the guy who made the original mod updated it on his page for 2.0+. I'm discontinuing working on this here as a result, though you could direct this question towards them and they might be able to add this functionality to the mod
