small edits in code formatting