Add condition for draw