Add in 2 more tests