Update

Improvised on the for loop. No more messy nested for loop from line 55