Modify trace api

@mortenvp I finished the unit tests and fixed the todos as good as i could. Please let me know what you think? :)
So just to be clear I think it should be `input_symbol_coefficients` in both places