Update class for gallery