Almost got the update to work with modal. not quite working