Latest code has been updated

Finished coding the extraction of test samples regardless of the value
of N