simple_example.py should be working with the refactored code