Drawing pattern using python