Milestone write up and some fixes to backpropagation