Good job isolating this responsibility into one class!