V_unbound=np.array([V_well(i) for i in x])
E,psi_unbound=solve(x,V_unbound)
E /= eV


# plot probability functions
plt.title('Single well, unbound states', fontsize=20) #THIS SEEMS TOTALLY OFF TOO

for i in range(4,7): # Three last states [4,7)
    psi_n = psi_unbound[:,(i-1)]
    #psi_n = psi[:,(i-1)]
    plt.plot(x_nm, psi_n**2, 'rgbmyc'[(i-1)%6], label='$|\Psi_{%d}|^2$ (num)' % (i))
plt.legend()
plt.xlabel('$x$ (nm)',fontsize=15)
plt.ylabel('$|\psi|^2$',fontsize=15)
#plt.xlim((-Nw*(w+g)*pow(10,10),Nw*(w+g)*pow(10,10)))
plt.show()