Improve notification button UI