Added documentation to what I changed in Sohum's code