Add media entity support