Matrix Program In Python User Input