Getting ready for the first test

added for loops, defined variables
Trying to find the right indices

trying to get rid of this:
IndexError: arrays used as indices must be of integer (or boolean) type
Added print commands
minor changes in hope to fix it..