Adding notification code

Adding a new class and an edited class for creating notifications.