More pythonic way to set polygon points