Modify cookbook so it's more flexible