Add subscribe feature
Add test for subscribe