Final report edits and code cleanup