Rydda opp i draw-funksjonen