Update dist.py

Added import statement for convenience