Clean up and prepare for AABB on GPU