Updated notifications and added proper tests