A Python solver for the 1D Schrodinger equation