Basic Swift drawing code