The edit was to just make it look prettier.

I'll look at it again after our theoretical test