Add logging & TODO