Draft mode bug
Amend TODO