Add some custom stuff