edited the return types