Restore ability to run with python -m