Python If Else Hackerrank Solution For Electronics