Added more code to restaurant system