Refs #9 - Updates tests.