Tom has finally had a chance to have a go at the code after a long while and has fixed up a number of things in the package. It is highly recommended that anyone using previous versions of the package upgrade to the latest version and retrain their models.