Added rollback() in try/except