Super simplified 

-tightened up the loop
-got rid of class and method