Add drawing code