Add subscribe and unsubscribe functionality