Add dynamic styling and save function