add test input function