@Override
    public void add(int index, E element) {
        int count = 0;
        Node<E> prev = null;

//        if (!(index > size())) {

        if (index == 0) {
            Node<E> newNode = new Node<>(element);
            newNode.next = first;
            first = newNode;
        } else {
            for (Node<E> temp = first; temp != null; temp = temp.next) {
                if (count == index) {
                    Node<E> newNode = new Node<>(element);
                    prev.next = newNode;
                    newNode.next = temp;
                }
                count++;
                prev = temp;
//                if (temp.next == null) {
//                    last = temp;
//                }
            }
        }
        if (index == size()) {
            Node<E> newNode = new Node<>(element);
            prev.next = newNode;
            last = newNode;
        }
//        }
    }