Picking Numbers Hackerrank Solution In Python