added notifications to correct function