Add visualisation of joints