Address @jasongrout 's review comments
Use load_class in comms instead of callbacks.