Add image class in partial