Added missing code snippet