Normalize weights in feedforward better

Each layer should sum to 1

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>