Small changes

Small cahnges