started expanding the snippet to account for tests