Fixed minor bug with model convergence. Updated requirements.txt