Yes, you can. When buying the ticket and specifying the number of tickets, press BUY RETURN TICKET* and you will be able to select your return trip. Then you just need to know the departure time of your return trip train.
*Please note that starting from April 1, 2024, the discount is no longer applied to the return ticket.