Use a context manager to manage drivers.

All tests passing. Closes #78.