STY: Update based on forward functions