added minor todo