simpify test, add TODOs