added notifications and UI improvements