Rewrite part of snapshot feature to get connections correctly

I've tried to minimize the amount of code I'm touching to complete this. I'd prefer not to rewrite the entire class.
Yep, I whipped this out last night. Are you okay with naming it something random? This is how sspak does it.