Hello,
you can buy a return ticket on the SBB Mobile App by clicking on the double arrow after selecting the outward journey (the double arrow appears below the passenger data).
Note, however, that in this way the return journey will be purchased for the same day as the outward journey. If you have to return on a different date, please buy the outward journey first and then the return.
I wish you safe travels,
Chiara_S