fixed imports in classes.py