Updated Main Unit

Updated Main Unit to associate datasets correctly

Signed-off-by: donn <DonnOMalley1030@gmail.com>