Add generate and restore options