Added world_gen.py. Generating different worlds for testing agents.