Update init.lua

Just a Suggestion.

Also, I'm not to sure if the if statement is correct.