Adding missing code.