Make config.py more versatile