Fixed Diff!

Now onto better tests