Got it to sort of work