Using Python to write faster code