Adding the draw code.