Minor changes/comment addition