minor edits to layout of code