Added more tests to show off the functionality