Made add() work for both lof and light_lof