updated TODO
Added option to choose layers activation functions