add part of test code