made a couple of changes

made a couple of changes