fixed todo code