Update __init__.py

add keyword class