Add ConnectionCallback interface