Move some classes to a separate Python package