add methods to __init__.py