Improve notification led code

It actually works now like original does.