Added some simple functionality to test.