taking integer input in python