Python Script to Solve Schrodinger Equation for input potential