Fixed 'Undo' functionality
Fixed grunt build