Image rendering customisation

I understand Contexts are the way to go to get custom rendering of Image atoms;
This provides a lot of settings for the content of the atom, but not for the outer div (dnd-atom-wrapper);
In order to provide the very basic custom contexts (like float-left, float-right, framing,...) which are easy to do with plain ckeditor images, I need a context-specific class for the atom wrapper...
I suggest replacing in the Default atom theme function, scald/modules/library/dnd/js/dnd-library.js line 138 by :

//var output = '<div class="dnd-atom-wrapper" data-scald="' + encodeURIComponent(sas) + '"><div class="dnd-drop-wrapper">' + atom.contexts[context] + '</div>';
  var laClasse = 'dnd-atom-wrapper dnd-atom-wrapper-' + context;
  var output = '<div class="' + laClasse + '" data-scald="' + encodeURIComponent(sas) + '"><div class="dnd-drop-wrapper">' + atom.contexts[context] + '</div>';

Thanks,