Small TODO comment added