Functions

$f(x) = (x-1)^2$ for $x \geq 1$, $g(x)$ is a function whose graph is the reflection of the graph of $f(x)$ in the line $y = x$, then $g(x)$ is