Add ability to close connections