Managed to get it working