Clean up example with imports and break statement