Cleaned up main a bit and new storage structure for proabilities needed in Cohort generation