Update to use wordpress styles and js loader function