Circuit Python If Else Hackerrank