Программирование attiny2313 на C++

Написал код на C++ для мк attiny2313 хотел бы узнать как этот код должен выглядеть на самом деле, потому что этот я полагаю костыль тот еще

Суть кода: при нажатии на кнопку присоедененную к порту А 0b00000001 подается логическая 1 последовательно на порты В 0b00000001 - 0b00000100 когда счетчик достигает лимита, в данном случает 3, то все обнуляется и начинается заново.

#include <avr/io.h>

int main(void)
{
    DDRB = 0xff;
    int count = 0;
    while (1) 
    {
        if (PINA == 0b001)
        {
            while(1)
            {
                if (PINA == 0b000)
                {
                    break;
                }
            }
            if (count <= 0)
            {
                count++;
                PORTB |= 1<<1;
            }
            else
            {
                count = 0;
                PORTB = 0b00000000;
            }
        }
    }
}