Added tests that were somehow missed in transition

:(