Fix draw example.

Improve this example as mentioned in #344