Will probably add more tests later