diff --git a/README.md b/README.md
index 7790ea741610b4e9cb9757a32860e359441de8b3..6ad3f9484f231a7c0a39edbbb662657da16d0463 100644
--- a/README.md
+++ b/README.md
@@ -158,12 +158,19 @@ Edit `AppDelegate.m`:
 + - (void)userNotificationCenter:(UNUserNotificationCenter *)center willPresentNotification:(UNNotification *)notification withCompletionHandler:(void (^)(UNNotificationPresentationOptions))completionHandler
 + {
 +   [[NSNotificationCenter defaultCenter] postNotificationName:FCMNotificationReceived object:self userInfo:notification.request.content.userInfo];
-+   completionHandler(UNNotificationPresentationOptionAlert);
++     if([notification.request.content.userInfo valueForKey:@"show_in_foreground"]){
++     completionHandler(UNNotificationPresentationOptionAlert | UNNotificationPresentationOptionBadge | UNNotificationPresentationOptionSound);
++   }else{
++     completionHandler(UNNotificationPresentationOptionNone);
++   }
+
 + }
 
 + - (void)userNotificationCenter:(UNUserNotificationCenter *)center didReceiveNotificationResponse:(UNNotificationResponse *)response withCompletionHandler:(void (^)())completionHandler
 + {
-+   [[NSNotificationCenter defaultCenter] postNotificationName:FCMNotificationReceived object:self userInfo:response.notification.request.content.userInfo];
++     NSDictionary* userInfo = [[NSMutableDictionary alloc] initWithDictionary: response.notification.request.content.userInfo];
++   [userInfo setValue:@YES forKey:@"opened_from_tray"];
++   [[NSNotificationCenter defaultCenter] postNotificationName:FCMNotificationReceived object:self userInfo:userInfo];
 + }
 + #else
 + //You can skip this method if you don't want to use local notification