Removed a few bugs from the 2d.py file. Updated input and output values to better reflect the expected behaviour.