add argument for test