Added notification feature