First test almost working, probably problem with data shapes.

First test almost working, probably problem with data shapes. #51