bugfix: dimension of A in toy_example.py