Lagt till ett par TODOs rörande balance-handling