Modify image class