How To Put Constraints On Integer Input In Python