nf $=$ input$("$Final state $($nf$)$: $")$

nf $=$ int$($nf$)$

ni $=$ np$.$arange$($nf$+1,$nf$+$n$+1)$

plt$.$plot$($ni$,$nist,$\text{'}$bx$\text{'})$

plt$.$grid$($True$)$

plt$.$show$()$