Tidy up micromag flux elements

* Comments

* Remove useless functions