adding a proper return