Yes, it would work also.

I am just usually too lazy to do some classes when I can see solution with HashMap :)