now using sigmoid to predict. Doing a lot better