Add some draft functionality