Python Shape Pattern