Wawrinka wants improvement despite win