Update restaurant.py