add comment, small improvement in code