Custom extension for keystone