Energy changes in a falling ball | StemPy Lab