#43 - add cnn specifc classes