Add class for image blocks