Enhance code

This way the diff repos can be updated as well