small changes

still needs to be done