Add proper return values