Close the connection properly on success, too

I *knew* Pythons significant whitespace would bite me some day. It's one
of the reasons I don't really like Python = S