Separation of output layer
Merge pull request #12 from jakubsacha/master

Separation of output layer