Issues with looping - Need help refining my code