Small changes added