Minor Edit: Added bone structure to setTokenClass

Added some comments and if statements as a backbone structure.