Add update to both functions, test getController in the update function