@f3n1Xx did you decide to add features to your classificaton layer? This is what I am working on. @nielsr do you think this method would have better results?