Wakefield Trinity keep pace with the top few sides in the Super League table following a comfortable win at Castleford Tigers.