training and tag functionalities done! check post processing function and baseline features