cool shapes using python