Added two tests, Yay