add tests for Story 4, and will continue on adding tests