Added a draft button. Fixes #4