Add surface_snapshots to the setup.py file