Python/MATLAB Problem Solving