Started working on the tests by copying old code. Make sure to modify this.