Working forward prop with corner case

Back Prop is started - still needs to be finished and then one
epoch of training will be done.