Add 2 todos and fix 2 bugs