made a few small changes