Update remove media file function