Added test for story method and refactored test suite