Redesign notifications class