Better testing, added loops for accurate testing on set data