MAde it simpler to change use case in main.
minor bug fixes in policygenerator
removing cout