Small changes

small changes in code.