Add support for export function