Code edited

Changed main method, added for loop. Included name in author