Add reset function

Reset function is slightly broken as one of the elements disables
itself. Give it a test for yourself.