Wawrinka admits he must improve