Made storybook prettier