joint models and add visualizing tools