tweak main.py and documentation