Updated while loop

While loop second try :D