Added initial unit testing.