Adding rollback_all and migrate_all