adds return declaration back in