Added media removal functionality