test updates
loss function 1 bug
documentation