Olly Stone impresses as England take five Sri Lanka wickets before tea

England had earlier collapsed from 221 for three overnight to 325 all out in the space of just two hours.