Fix balance sheet bug
Update TODO