Adding tests
tests now passing for the right reason