Update rigid_body.cpp