Add in modification notification and remove unused import