We present the measurement of the differential cross-section of the $X(3872)$ state through its decays to $J/\psi \pi^+ \pi^{-}$ final state . The cross-section was extracted for both prompt and non-prompt production. The existence of the $X(3872)$ suggests the presence of its bottomonium counterpart $X_b$. Search for $X_b$ with the ATLAS experiment in several final states, including $\Upsilon \pi^+ \pi^{-}$, is presented.