added support for using stichastih gradient descent, does not work well yet. Should implement storage of linear parameters to reduce comp. cost