Add subscribe functionality