Add token related stuff to User model