Fix the draw function (homework, star)