Made code a lot nicer with array/loop