improving restaurant code