Improved Callaway model for lattice thermal conductivity