Fixes for adding class to notes