Added code to NotificationViewController