Add token check