I almost got it working