Please add more test cases and organized..