fix output, add todo