Cleaned up init.lua