Clean up connection code and use celluloid on the node.