Modified "remove" code example