Adjust UI for the media/image