made some small changes