modified the restaurants code too