small edits in code neatness