Added Brain, retrained, etc
Commented unnecessary code