Adding in a unittest