Actually got it working