Update world_gen.py

Added pygame display because pretty