Codility Lesson 1 : Binary Gap

def solution(N):
    binary = format(N, "b")
    zeros = 0
    count = 0

    for i in range(len(binary) - 1) :
        if "1" in binary[i+1:] and int(binary[i]) == 0:
            count += 1
        if count > zeros:
            zeros = count
        if binary[i] == "1":
            count = 0

    return zeros

def main():
    solution(1041)

if __name__ == '__main__':
    main()