Add image class framework