Writing a DSL with Python