Add tests for subscribe/unsubscribe
Add unsubscribe function