Draft mode fixed

Closes #34.