finished while loop. Need to fix ==