- better assert messages
- added softmax layer