did some small changes