remove test code
add flag