Add todo and fix formatting