Changed visuals from being a class to a set of functions