Return ticket to save the risk, the yanks will get concerned about you flight training and will want a quick chat in the back room. Well I had that but I guess its because I was there for a year instead of a month.
Why go there risking it? Your coming home anyway so get a return ticket but put it to the end of the Visa if you finish early then change it. Simplez