remove states import from __init__.py