Update plugin to work with new page code component structure