Mason Mount is back to training after being on medication for two weeks for the surgery on his wisdom teeth