Almost workingm just trying to fix an issue with input weights not being updated correctly.