Perfect answer. Wrapping the entire script in a function is a clever idea to protect against connection drops as well.