Trying to get this to work!! Doing while loop instead of if.