getting about 85% on test set