Added registering shape for visualisation.