Added a couple test