changed draw function and added comment describing further what it should do