Update units list and variables in the AE dataset to include units