Added imports and changed notification message