Added test code.

Added test code. Second commit.