update to reflect code