how to draw rectangle in python