HW5: Added a test