Some small changes to rigidbody