Added some comments and cleaned-up the code for the natural gradient.