Refactored revert-method
Merge pull request #229 from oskarols/revert-refactoring

Refactored revert-method