Modular code with Python