Added command line argument for notification feature