Added image class to repository