Updated controllers with injections