Add TODOs to ugly code