bugfixes
snapshot 1 (train and experiments 1,2 success)