Cleanup.

- Using a with statement.