Clean up argument checking in example.lua