Refactor keywords to own token keys

As discussed. This should hopefully improve performance and make the
parser easier to implement.