Set up what I'm pretty sure is the correct drawing code