Novak Djokovic trained his brain like this