add time step restriction;
add tests
use mass fraction as out_puts