Cleaned up main.lua

The for-loops worked this time :( Not sure why it didn't let me before.