Bug fixes

The smart_display function works properly now.
Solving an import error.