Stanislas Wawrinka hevet seg fra første kamp i Australian Open og slo Steve Johnson i strake sett i andre runde.