Fixed some matrix dimensionality errors
Changed num_samples to 1 for testing purposes