Add debug menu code