move step(..., n) to a separate file to make it more obvious you're meant to customize that function.