Found some more stuff I needed to add to get it working.