Revert "Remove rollback functionality. Fixes #4"

This reverts commit 86b788fda907db3a1441d53c67e4c2154d125dc9.