Small Fixes/Changes

Did more testing, revamped the testing loop. Also added a few things to
the arffWrapper. Still can't figure out why adding the gradient is
reducing the error