Notification almost working properly