Got it working. Still implementing