Logistic + Softmax + better inheritance
Setup.py + better import structure