Add rollback capabilities