Add context manager to close database connection
Remove debug statements
Merge pull request #2 from sot/context-manager

Add context manager