Adding function types as return types