Trainable version of 2-layer CNN
added untracked files