Small changes after correcting