Improving pupils' short-term memory boosts maths results, study finds