Full code review

Vocabular size ?