Updating Python test with a loop

Just playing around with updating things