small edits to respect basic code styling