Update init.lua

this is a better version i think