Added tests and updated the package