Refactor code

As specified in README, there are some dubious design issues (there are some 'historical' motivations, but it doesn't matter). The code should be refactored to be more intuitive. 

This is low priority, I would rather prefer to extend it at least with convolutional networks and not fix something that's not broken.