small comments about solution

small comments about solution