small fix plus comments
fix comment