Small changes

Small changes