Simple illustration of Hamiltonian Monte Carlo in Python