small changes

Small changes