WIP: Get example code to work in a rake task