can you re-use the n_outputs here.