Minor tweaks

So the diff module partially works, but it needs… a lot of work :(