cnn_cost runs through now, still need to test correctness though