Add test code. Working.