started changing drawing code to include an arrow!