- Added rigid body generation to the model