I added tests with 0 in addition to 1