Refactors update ops for better extensibility and adds  support.