Made changes to rigidbody module