Adding test to Pegasos'