Get voxel prediction working