add missing code to repo