worked a bit on linear classifier