Fixed for python3 input