Make overriding/wrapping the draw/init calls more convenient.

/cc @mikoalysenko