Монету бросают пять раз. Найти вероятность того, что «герб» выпадет: а) менее двух раз; б) не менее двух раз

if n == 1:
        return 1
    else:
        return n * factorial(n-1)
 
def sochetania(n,k):
    s = int((factorial(n) / (factorial(k)* (factorial(n-k)))))
    #s = ((factorial(n+k-1))/(factorial(k)*(factorial(n-1))))
    return s
 
def main():
    n=10 #input('Введите количество попыток: ')
    k=5 #input('Введите желаемые исходы: ')
    n1 = 52 #input('Введите общее количество исходов для расчета вероятности: ')
    m = 4 #input('Введите количество интересующих исходов для расчета вероятности : ')
    p = m/n1
    B = (sochetania(n,k)) * (p**k) * ((1-p)**(n-k))
    print(B)
 
main()

from math import factorial
 
def sochetania(n, k):
    s = int((factorial(n) / (factorial(k) * factorial(n-k))))
    return s
 
def main():
    n = 5 # Количество подбрасываемых монет
    k = 2 # Количество выпадений герба
    n1 = 2 # общее количество исходов для подбрасывания монеты (орел или решка)
    m = 1 # Количество желаемых результатов (герб)
    p = m/n1 # Вероятность получения желаемого результата (герба) за один бросок
    
    # a) Вероятность получения герба меньше чем в два раза
    less_than_twice = 0
    for i in range(0, 2):
        less_than_twice += (sochetania(n, i) * (p**i) * ((1-p)**(n-i)))
    print(f"Вероятность получения герба меньше чем в два раза: {less_than_twice}")
    
    # b) Вероятность получения герба как минимум дважды
    at_least_twice = 1 - less_than_twice
    print(f"Вероятность получения герба как минимум дважды: {at_least_twice}")
 
if __name__ == "__main__":
    main()