* Remove batch normalization * Minor fixes * Fix typo * Fix comments * Add assertion for use_feat_batchnorm