Currently, there are no tests, and hence no code coverage.
We should test the algorithms (at least 1 or 2 epochs with a small fas) to check if the forward pass and parameter updates are working properly.
This can be tackled after #26 is merged.
cc: @veds12
