Adding code to move shapes in 3 dimensions.