Minor bug fix and updated versioning