Added functionality to receive notifications