Working! Just need to use rand to generate test data.
Cleanup tests.. Now to start making a class of it, and add regularization etc.