disc.py now works also in 2d