added comment and small fix