Updated snippet to reflect code