fixed test/train split and baseline function