Made an error in the code. After testing it appears to be working flawlessly with this new version. Please let me know if the current version doesn't work for you. Apologies for needing to update what should have been a very easy and small mod.