Modified the case for :notice and added a case for :success