Python While Loop With User Input Matlab