test is working, add more cases soon