Fix large image problem by adding class