small change/update in code neatness