added a bit more to the back prop