Got it sorta working