Added rollback

Attempt 2
Merge pull request #4 from Wild1145/patch-3

Added rollback