Split up token.py
Organize imports