Little fix + todo update