Adding media remove functionality for entity deletion