Add center of mass tracking