updated version for sending to pypi
remove unnecessary reimport