Add notification

Added notification while flashlight is on.

Some minor code changes too.