Maybe this could be passed as an option to the class?