Add comments, reformat code, update test code