https://researchdatapod.com/how-to-solve-python-valueerror-operands-could-not-be-broadcast-together-with-shapes/