almost there, trying to fix back prop