Iterations added, need to change for loop to while and add buttons