You can consolidate these lines:

    from lettuce import step, world, before