During past few days I started to work on small UX improvement which allows you to assign/unassign puppet classes to a host/hostgroup by clicking on it's name. You can still use the small icon on particular line but I find this more useful and less prone to errors.

I'll attach PR info when ready.