Внес изменения

Изменил метод добавления последнего элемента, добавил тестов, чуть
сократил код