added tests and a wrapper for the py