Adds basic test code