add test module logic