Add the media part and update the album controller