Add activation function

It closes #62