Add a way to close the connection