Update in test bed and __init__.py