Refactor take_step
Implement zero bias linear layer
add TLin an TRec layers
Fixed sqrt missing on first update
messing around with TRec and TLin on Cifar-10