Issue #6: Made widget class setup in auto mode