How to take integer input in python - YouTube