Found more unit tests to expand on :)