Adding rollback for transaction handling