Added CVAE but not tracking it