Add tests to cover new methods and add new methods to __init__.py