simpler update model across all layers