Katarina Johnson-Thompson believes her joint bronze medal at the World Championships in Tokyo gifted her the opportunity to write a "better story" about the city where she endured a "big trauma" four summers ago.