fixed inertia display to work with all models
merged changes to inertia viz