Modify database rake task