Added a While loop in that son bitch