Added a story class
Fixed comments