Corrected bounce.py and created working version of 2D_simple_bounce.py