Added more test for exercise