works installed locally without pip install -e .