Commiting changes to gen_gallery.py