Add back in raymarching function