code edits for prettiness