fixed package name
Fixed flake8 complains