add test for draw