Add restaurant logic