It works! Add in the media plugin and rework JS to use that