Подправил классы, добавил equals() и hashCode()