made a couple of small changes