Checkout the MMpolicygen from branch MultiModal