Adding ability to add callback for initial connection event