Britney Spears and Kevin Federline have settled their custody battle – avoiding a messy trial.

According to the agreement, reportedly signed by all parties, Federline gets sole custody of the couple’s two sons, Preston, 2 1/2, and Jayden, 1 1/2. Spears will have visitation rights.

“The case has been settled,” Federline’s attorney, Mark Vincent Kaplan, told E! Online. “The court still has to approve it. ” The two are due in court Friday. -people