Reformat code for unit testing lab.