Add missing return types