Think I got it working!