Improve notification menu