Added feature analysis and multiple SGD runs