Got shaky rollback working. Needs to be improved